import io.kuzzle.sdk.*;
import io.kuzzle.sdk.Options.KuzzleOptions;
import io.kuzzle.sdk.Options.Protocol.WebSocketOptions;
import io.kuzzle.sdk.Protocol.WebSocket;

import java.util.concurrent.ConcurrentHashMap;

public class mCreateDocument {
    private static Kuzzle kuzzle;

    public static void main(String[] args) {
        WebSocketOptions opts = new WebSocketOptions();
        opts.setAutoReconnect(true).setConnectionTimeout(42000);

        try {
            WebSocket ws = new WebSocket("localhost", opts);

            kuzzle = new Kuzzle(ws, (KuzzleOptions) null);

            kuzzle.connect();

            ConcurrentHashMap<String, Object> document1 = new ConcurrentHashMap<>();
            ConcurrentHashMap<String, Object> body1 = new ConcurrentHashMap<>();

            ConcurrentHashMap<String, Object> document2 = new ConcurrentHashMap<>();
            ConcurrentHashMap<String, Object> body2 = new ConcurrentHashMap<>();

            document1.put("_id", "some-id1");
            body1.put("key", "value");
            document1.put("body", body1);

            document2.put("_id", "some-id2");
            body2.put("key", "value");
            document2.put("body", body2);

            ArrayList<ConcurrentHashMap<String, Object>> documents = new ArrayList<>();
            documents.add(document1);
            documents.add(document2);

            ConcurrentHashMap<String, Object> response =
            kuzzle.getDocumentController().mCreate("nyc-open-data", "yellow-taxi", documents)
            .get();

            System.out.println(response);
        }  catch (Exception e) {
            e.printStackTrace();
        }

        kuzzle.disconnect();
    }
};